package com.zhang.common.core.utils.ip;

import com.zhang.common.core.utils.StringUtils;
import jakarta.servlet.http.HttpServletRequest;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.InetAddress;
import java.net.UnknownHostException;

/**
 * @Class com.zhang.common.core.utils.ip
 * @Author zhang
 * @Date 2025/9/15 11:07
 * @Description 获取IP方法
 **/
public class IpUtils {

    private static final Logger log = LoggerFactory.getLogger(IpUtils.class);

    /**
     * 获取客户端IP
     *
     * @param request 请求对象
     * @return IP地址
     */
    public static String getIpAddr(HttpServletRequest request) {
        // 请求对象判空
        if (request == null) {
            return "unknown";
        }
        // 获取 IP
        String ip = request.getHeader("x-forwarded-for");
        // 检查代理服务器专用头部 Proxy-Client-IP（常用于 Squid、Apache 等传统代理）
        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("Proxy-Client-IP");
        }

        // 检查标准化头部 X-Forwarded-For（多级代理时会包含多个IP，第一个为真实IP）
        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("X-Forwarded-For");
        }

        // 检查 WebLogic 服务器的特定头部 WL-Proxy-Client-IP
        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("WL-Proxy-Client-IP");
        }

        // 检查 Nginx 等现代代理的头部 X-Real-IP（通常直接包含客户端IP）
        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getHeader("X-Real-IP");
        }

        // 最终回退: 若所有代理头部均未获取到有效IP，使用底层连接的远程地址
        if (ip == null || ip.isEmpty() || "unknown".equalsIgnoreCase(ip)) {
            ip = request.getRemoteAddr();
        }

        // 返回获取的 IP
        return "0:0:0:0:0:0:0:1".equals(ip) ? "127.0.0.1" : getMultistageReverseProxyIp(ip);
    }

    /**
     * 检查是否为内部IP地址
     *
     * @param ip IP地址
     * @return 结果
     */
    public static boolean internalIp(String ip) {
        byte[] addr = textToNumericFormatV4(ip);
        return internalIp(addr) || "127.0.0.1".equals(ip);
    }

    /**
     * 检查是否为内部IP地址
     *
     * @param addr byte地址
     * @return 结果
     */
    private static boolean internalIp(byte[] addr) {
        if (StringUtils.isNull(addr) || addr.length < 2) {
            return true;
        }
        final byte b0 = addr[0];
        final byte b1 = addr[1];
        // 10.x.x.x/8
        final byte SECTION_1 = 0x0A;
        // 172.16.x.x/12
        final byte SECTION_2 = (byte) 0xAC;
        final byte SECTION_3 = (byte) 0x10;
        final byte SECTION_4 = (byte) 0x1F;
        // 192.168.x.x/16
        final byte SECTION_5 = (byte) 0xC0;
        final byte SECTION_6 = (byte) 0xA8;
        switch (b0) {
            case SECTION_1:
                return true;
            case SECTION_2:
                if (b1 >= SECTION_3 && b1 <= SECTION_4) {
                    return true;
                }
            case SECTION_5:
                if (b1 == SECTION_6) {
                    return true;
                }
            default:
                return false;
        }
    }

    /**
     * 将IPv4地址转换成字节
     *
     * @param text IPv4地址
     * @return byte 字节
     */
    public static byte[] textToNumericFormatV4(String text) {
        // IPv4 地址判空
        if (text.isEmpty()) {
            return null;
        }

        byte[] bytes = new byte[4];
        // 使用 -1 保留所有空元素（避免省略末尾空值）
        String[] elements = text.split("\\.", -1);
        try {
            // 用于存储解析后的数值
            long l;
            // 循环索引
            int i;
            // 根据分割后的元素数量处理不同IP格式
            switch (elements.length) {
                // 单数值格式（32位整型格式，如 3232235521）
                case 1:
                    l = Long.parseLong(elements[0]);
                    // 验证 32 位无符号整数范围：0 ~ 2^32-1（4294967295）
                    if ((l < 0L) || (l > 4294967295L)) {
                        return null;
                    }
                    // 按大端序分解为4个字节（位移+掩码操作）
                    bytes[0] = (byte) (int) (l >> 24 & 0xFF); // 取最高8位
                    bytes[1] = (byte) (int) ((l & 0xFFFFFF) >> 16 & 0xFF); // 次高8位
                    bytes[2] = (byte) (int) ((l & 0xFFFF) >> 8 & 0xFF); // 次低8位
                    bytes[3] = (byte) (int) (l & 0xFF); // 最低8位
                    break;
                // 双段格式（如 192.168 等价于 192.0.0.168）
                case 2:
                    // 处理前半段（首个字节）
                    l = Integer.parseInt(elements[0]);
                    // 验证字节范围
                    if ((l < 0L) || (l > 255L)) {
                        return null;
                    }

                    // 直接存储第一个字节
                    bytes[0] = (byte) (int) (l & 0xFF);
                    l = Integer.parseInt(elements[1]);
                    if ((l < 0L) || (l > 16777215L)) {
                        return null;
                    }
                    // 分解为第二字节
                    bytes[1] = (byte) (int) (l >> 16 & 0xFF);
                    // 第三字节
                    bytes[2] = (byte) (int) ((l & 0xFFFF) >> 8 & 0xFF);
                    // 第四字节
                    bytes[3] = (byte) (int) (l & 0xFF);
                    break;
                // 三段格式（如 10.10.12345 等价于 10.10.48.57）
                case 3:
                    // 处理前两段（每个必须为单字节）
                    for (i = 0; i < 2; ++i) {
                        l = Integer.parseInt(elements[i]);
                        // 单字节校验
                        if ((l < 0L) || (l > 255L)) {
                            return null;
                        }
                        bytes[i] = (byte) (int) (l & 0xFF);
                    }

                    // 处理第三段（双字节组合数值）
                    l = Integer.parseInt(elements[2]);
                    // 验证16位数值范围（0~65535）
                    if ((l < 0L) || (l > 65535L)) {
                        return null;
                    }
                    // 分解为第三字节
                    bytes[2] = (byte) (int) (l >> 8 & 0xFF);
                    // 第四字节
                    bytes[3] = (byte) (int) (l & 0xFF);
                    break;
                // 标准四段格式（如 192.168.1.1）
                case 4:
                    for (i = 0; i < 4; ++i) {
                        l = Integer.parseInt(elements[i]);
                        // 每个必须为单字节值
                        if ((l < 0L) || (l > 255L)) {
                            return null;
                        }
                        // 直接存储
                        bytes[i] = (byte) (int) (l & 0xFF);
                    }
                    break;
                // 无效格式（分割后元素数量不符合任何合法形式）
                default:
                    return null;
            }
        } catch (NumberFormatException e) {
            return null;
        }
        // 返回成功转换的4字节数组
        return bytes;
    }

    /**
     * 获取IP地址
     *
     * @return 本地IP地址
     */
    public static String getHostIp() {
        try {
            return InetAddress.getLocalHost().getHostAddress();
        } catch (UnknownHostException e) {
            log.error("获取本地 IP 异常：{}", e.getMessage(), e);
        }
        return "127.0.0.1";
    }

    /**
     * 获取主机名
     *
     * @return 本地主机名
     */
    public static String getHostName() {
        try {
            return InetAddress.getLocalHost().getHostName();
        } catch (UnknownHostException e) {
            log.error("获取主机名异常：{}", e.getMessage(), e);
        }
        return "未知";
    }

    /**
     * 从多级反向代理中获得第一个非unknown IP地址
     *
     * @param ip 获得的IP地址
     * @return 第一个非unknown IP地址
     */
    public static String getMultistageReverseProxyIp(String ip) {
        // 多级反向代理检测
        if (ip != null && ip.indexOf(",") > 0) {
            final String[] ips = ip.trim().split(",");
            for (String subIp : ips) {
                if (!isUnknown(subIp)) {
                    ip = subIp;
                    break;
                }
            }
        }
        return ip;
    }

    /**
     * 检测给定字符串是否为未知，多用于检测HTTP请求相关
     *
     * @param checkString 被检测的字符串
     * @return 是否未知
     */
    public static boolean isUnknown(String checkString) {
        return StringUtils.isBlank(checkString) || "unknown".equalsIgnoreCase(checkString);
    }


}
